/*
 * Copyright (c) 2016, ARM Limited and Contributors. All rights reserved.
 *
 * SPDX-License-Identifier: BSD-3-Clause
 */

#ifndef __CONTEXT_H__
#define __CONTEXT_H__

/*******************************************************************************
 * Constants that allow assembler code to access members of and the 'regs'
 * structure at their correct offsets.
 ******************************************************************************/
#define CTX_REGS_OFFSET        0x0
#define CTX_GPREG_R0        0x0
#define CTX_GPREG_R1        0x4
#define CTX_GPREG_R2        0x8
#define CTX_GPREG_R3        0xC
#define CTX_LR            0x10
#define CTX_SCR            0x14
#define CTX_SPSR        0x18
#define CTX_NS_SCTLR        0x1C
#define CTX_REGS_END        0x20

#ifndef __ASSEMBLY__

#include <cassert.h>
#include <stdint.h>

/*
 * Common constants to help define the 'cpu_context' structure and its
 * members below.
 */
#define WORD_SHIFT        2
#define DEFINE_REG_STRUCT(name, num_regs)    \
    typedef struct name {            \
        uint32_t _regs[num_regs];    \
    }  __aligned(8) name##_t

/* Constants to determine the size of individual context structures */
#define CTX_REG_ALL        (CTX_REGS_END >> WORD_SHIFT)

DEFINE_REG_STRUCT(regs, CTX_REG_ALL);

#undef CTX_REG_ALL

#define read_ctx_reg(ctx, offset)    ((ctx)->_regs[offset >> WORD_SHIFT])
#define write_ctx_reg(ctx, offset, val)    (((ctx)->_regs[offset >> WORD_SHIFT]) \
                     = val)
typedef struct cpu_context {
    regs_t regs_ctx;
} cpu_context_t;

/* Macros to access members of the 'cpu_context_t' structure */
#define get_regs_ctx(h)        (&((cpu_context_t *) h)->regs_ctx)

/*
 * Compile time assertions related to the 'cpu_context' structure to
 * ensure that the assembler and the compiler view of the offsets of
 * the structure members is the same.
 */
CASSERT(CTX_REGS_OFFSET == __builtin_offsetof(cpu_context_t, regs_ctx), \
    assert_core_context_regs_offset_mismatch);

#endif /* __ASSEMBLY__ */

#endif /* __CONTEXT_H__ */
